#ifndef HB_DROPOUT_H
#define HB_DROPOUT_H
#include "layer_base.h"

class dropout_layer_t: public layer_t
{
public:
	tensor_t<bool> hitmap;
	float p_activation;

	dropout_layer_t( tdsize in_size, float p_activation );

	void activate( tensor_t<float>& in ) override;

	void fix_weights() override;

	void calc_grads( tensor_t<float>& grad_next_layer ) override;
};

#endif //HB_DROPOUT_H